package socket;

import java.io.*;
import java.lang.reflect.Array;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;

/**
 * 聊天室服务端
 */
public class Server {
    /**
     * 运行在服务端的ServerSocket主要完成两个工作:
     *      1.向服务端操作系统申请服务端口,客户端就是通过这个端口与ServerSocket建立链接
     *      2.监听端口,一旦一个客户端建立链接,会立即返回一个Socket.通过这个Socket就可以和该客户端交互了
     * <p>
     * 我们可以把ServerSocket想象成某客服的"总机".用户拨打电话到总机,总机分配一个电话使得服务端与你沟通.
     */
    private ServerSocket serverSocket;

    /*
       存放所有客户端输出流,用于广播消息.
     */
//    private PrintWriter[] allOut = {};
    private Collection<PrintWriter>allOut = new ArrayList<>();

    /**
     * 服务端构造方法,用来初始化
     */
    public Server() {
        try {
            System.out.println("正在启动服务端...");
            /*
               实例化SeverSocket时要指定服务端口,该端口不能与操作系统其他应用程序占用的端口相同,否则会抛出异常:
                  java.net.BindException:address already in use

               端口是一个数字,取值范围:0-65535;
               6000之前的端口不要使用,密集绑定系统应用和流行应用程序.
             */
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕.");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 服务端开始工作的方法
     */
    public void start() {
        try {
            while (true) {
                System.out.println("等待客户端链接...");
            /*
               ServerSocket提供了接受客户端链接的方法:
               Socket accept()
               这个方法是一个阻塞方法,调用后方法"卡住",此时开始等待客户端的链接,直到一个客户端链接,
               此时该方法会立刻返回一个Socket实例通过这个Socket就可以与客户端进行交互了.

               可以理解为此操作是接电话,电话没响就一直等.
             */
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端链接了.");
                //启动一个线程与客户端交互
                ClientHandler clientHandler = new ClientHandler(socket);
                Thread t = new Thread(clientHandler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }

    /**
     *    定义线程任务
     *    目的是让一个线程完成与特定客户端的交互工作
     */
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host;//记录客户端ip地址
        public ClientHandler(Socket socket){
            this.socket = socket;
            //通过socket获取远端计算机地址信息
            host = socket.getInetAddress().getHostAddress();
        }
        public void run(){
            PrintWriter pw = null;
            try{
                /*
               Socket提供的方法:
                 InputStream getInputStream()
                 获取的字节输入流读取的是对方计算机发送过来的字节
             */
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, "UTF-8");
                BufferedReader br = new BufferedReader(isr);

                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw,true);

//                synchronized (allOut){//不行,下面操作会扩容,allOut对象会变
//                synchronized (this){//不行,因为这个是ClientHandler实例

//                synchronized (Server.this){//外部类对象可以
                    //将该输出流存入共享数组allOut中
                    //1.对allOut数组扩容
//                    allOut = Arrays.copyOf(allOut, allOut.length + 1);

                /*
                    换成集合后,锁对象可以直接使用该临界资源allOut了,因为增删元素并不会创建一个新的集合赋值给allOut
                    (即:allOut对象本身没有发生过改变,只是内部元素增删而已.)
                 */
                synchronized (allOut) {
                    allOut.add(pw);
                    //2.将输出流存入数组最后一个位置
//                    allOut[allOut.length - 1] = pw;
                }

                //通知所有客户端该用户上线了
                sendMessage(host+"上线了,当前在线人数:"+allOut.size());

                String message = null;
                while ((message = br.readLine()) != null) {
                    System.out.println(host+"说:" + message);
                    //将消息回复给所有客户端
                    sendMessage(host + "说:" + message);
                }
            }catch (IOException e){
                e.printStackTrace();
            }finally {
                //处理客户端断开连接操作
                //将当前客户端的输出流从allOut中删除(数组缩容)
                synchronized (allOut) {
//                    for (int i = 0; i < allOut.length; i++) {
//                        if (allOut[i] == pw) {
//                            allOut[i] = allOut[allOut.length - 1];
//                            allOut = Arrays.copyOf(allOut, allOut.length - 1);
//                            break;
//                        }
//                    }
                    allOut.remove(pw);


                }
                sendMessage(host+"下线了,当前在线人数:"+allOut.size());

                try {
                    socket.close();//与客户端断开连接
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        /**
         * 广播消息给所有客户端
         * @param message
         */
        private void sendMessage(String message){
            synchronized (allOut) {
//                for (int i = 0; i < allOut.length; i++) {
//                    allOut[i].println(message);
//                }
                for (PrintWriter pw : allOut){
                    pw.println(message);
                }
            }
        }
    }
}